package com.ybg.oss;
import com.qcloud.cos.COSClient;
import com.qcloud.cos.ClientConfig;
import com.qcloud.cos.request.UploadFileRequest;
import com.qcloud.cos.sign.Credentials;
import com.ybg.base.util.webexception.ResultException;
import net.sf.json.JSONObject;
import org.apache.commons.io.IOUtils;
import java.io.IOException;
import java.io.InputStream;

/** 腾讯云存储
 * 
 * @author chenshun
 * @email sunlightcs@gmail.com
 * @date 2017-03-26 20:51 */
public class QcloudCloudStorageService extends AbstractCloudStorageService {
	
	private COSClient client;
	
	public QcloudCloudStorageService(CloudStorageConfig config) {
		this.config = config;
		// 初始化
		init();
	}
	
	private void init() {
		Credentials credentials = new Credentials(config.getQcloudAppId(), config.getQcloudSecretId(), config.getQcloudSecretKey());
		// 初始化客户端配置
		ClientConfig clientConfig = new ClientConfig();
		// 设置bucket所在的区域，华南：gz 华北：tj 华东：sh
		clientConfig.setRegion(config.getQcloudRegion());
		client = new COSClient(clientConfig, credentials);
	}
	
	@Override
	public String upload(byte[] data, String path) {
		// 腾讯云必需要以"/"开头
		String qclouduploadhead = "/";
		String jsonCodeName = "code";
		if (!path.startsWith(qclouduploadhead)) {
			path = "/" + path;
		}
		// 上传到腾讯云
		UploadFileRequest request = new UploadFileRequest(config.getQcloudBucketName(), path, data);
		String response = client.uploadFile(request);
		JSONObject jsonObject = JSONObject.fromObject(response);
		if (jsonObject.getInt(jsonCodeName) != 0) {
			throw new ResultException("文件上传失败，" + jsonObject.getString("message"));
		}
		return config.getQcloudDomain() + path;
	}
	
	@Override
	public String upload(InputStream inputStream, String path) {
		try {
			byte[] data = IOUtils.toByteArray(inputStream);
			return this.upload(data, path);
		} catch (IOException e) {
			e.printStackTrace();
			throw new ResultException("上传文件失败", e);
		}
	}
	
	@Override
	public String upload(byte[] data) {
		return upload(data, getPath(config.getQcloudPrefix()));
	}
	
	@Override
	public String upload(InputStream inputStream) {
		return upload(inputStream, getPath(config.getQcloudPrefix()));
	}
}